34
https://raw.githubusercontent.com/1127140426/tensquare/master/tensquare_friend/src/main/java/com/tensquare/friend/controller/FriendController.java
package com.tensquare.friend.controller;

import com.tensquare.friend.client.UserClient;
import com.tensquare.friend.service.FriendService;
import entity.Result;
import entity.StatusCode;
import io.jsonwebtoken.Claims;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletRequest;
import java.util.concurrent.TimeUnit;

/**
 * @author 李聪
 * @date 2020/2/19 10:59
 */
@RestController
@RequestMapping("/friend")
public class FriendController {

    @Autowired
    private FriendService friendService;

    @Autowired
    private HttpServletRequest request;

    @Autowired
    private UserClient userClient;

    @RequestMapping(value = "/like/{friendid}/{type}",method = RequestMethod.PUT)
    public Result addFriend(@PathVariable String friendid,@PathVariable String type) {
        //验证是否登录，并且拿到当前登录的用户id
        Claims claims = (Claims) request.getAttribute("claims_user");
        if(claims == null || "".equals(claims)) {
            //说明当前用户没有user角色
            return new Result(false,StatusCode.LOGINERROR,"权限不足");
        }
        //得到当前登录的用户Id
        String userid = claims.getId();
        //判断是添加好友还是添加非好友
        if(type != null) {
            if(type.equals("1")) {
                //添加好友
                int flag = friendService.addFriend(userid,friendid);
                if(flag == 0) {
                    return new Result(false,StatusCode.ERROR,"不能重复添加好友");
                }
                if(flag == 1) {
                    userClient.updatefanscountandfollowcount(userid,friendid,1);
                    return new Result(true,StatusCode.OK,"添加成功");
                }
            }else if(type.equals("2")){
                //添加非好友
                int flag = friendService.addNoFriend(userid,friendid);
                if (flag == 0) {
                    return new Result(false,StatusCode.ERROR,"不能重复添加非好友");
                }
                if(flag == 1) {

                    return new Result(true,StatusCode.OK,"添加成功");
                }
            }
            return new Result(false, StatusCode.ERROR,"参数异常");
        }else {
            return new Result(false, StatusCode.ERROR,"参数异常");
        }
    }
    @RequestMapping(value = "/{friendid}",method = RequestMethod.DELETE)
    public Result deleteFriend(@PathVariable String friendid) {
        Claims claims = (Claims) request.getAttribute("claims_user");
        if(claims == null || "".equals(claims)) {
            //说明当前用户没有user角色
            return new Result(false,StatusCode.LOGINERROR,"权限不足");
        }
        //得到当前登录的用户Id
        String userid = claims.getId();
        friendService.deleteFriend(userid,friendid);
        userClient.updatefanscountandfollowcount(userid,friendid,-1);
        return new Result(true,StatusCode.OK,"删除成功");
    }
}
